home *** CD-ROM | disk | FTP | other *** search
/ Amiga Plus 2004 #2 / Amiga Plus CD - 2004 - No. 02.iso / AmiSoft / Comm / tcp / JabberwockySRC.lha / Jabberwocky / src / http.c < prev    next >
Encoding:
C/C++ Source or Header  |  2003-04-06  |  7.8 KB  |  353 lines
  1. /*  Copyright (C) 2002 Tom Parker (tom@carrott.org),
  2.                        Matthias Münch (matthias@amigaworld.de)
  3.  
  4.     This program is free software; you can redistribute it and/or modify
  5.     it under the terms of the GNU General Public License as published by
  6.     the Free Software Foundation; either version 2 of the License, or
  7.     (at your option) any later version.
  8.  
  9.     This program is distributed in the hope that it will be useful,
  10.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.     GNU General Public License for more details.
  13.  
  14.     You should have received a copy of the GNU General Public License
  15.     along with this program; if not, write to the Free Software
  16.     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  
  18.     In addition, as a special exception, Tom Parker and Matthias Münch give
  19.     permission to link the code of this program with a TCP stack of your 
  20.     choice, any official MUI libraries or classes and any custom MUI classes
  21.     that should be necessary for the operation of this program.  This 
  22.     exception also gives you permission to distribute linked combinations 
  23.     including this software with any of the before-mentioned libraries and 
  24.     classes.  You must obey the GNU General Public License in all respects for
  25.     all of the code used other than that provided by the before-mentioned 
  26.     libraries and classes.  As part of this exception you are obliged to 
  27.     follow the license terms of the before-mentioned libraries, this license
  28.     does not compel you to follow those terms, but if you do not then you may
  29.     not link with those libraries.  If you modify this file, you may extend
  30.     this exception to your version of the file, but you are not obligated to
  31.     do so.  If you do not wish to do so, delete this exception statement from
  32.     your version.
  33. */
  34. /*
  35. ** built-in http implementation
  36. */
  37.  
  38. #include "common.h"
  39. #include "network.h"
  40. #include "madthread.h"
  41. #include <sys/stat.h>
  42.  
  43. extern struct Library *SocketBase;
  44.  
  45. struct connection
  46. {
  47.     listitem node;
  48.     struct xferdata *xf;
  49.     long key;
  50.     int state;
  51.     long sock;
  52.     int len;
  53.     char buf[4096];
  54. };
  55.  
  56. static struct fd_set fds;
  57. static struct timeval tv = { 0, 0 };
  58. static long sock = -1;
  59. static list *waitlist = NULL;
  60.  
  61. static char TryLater[] = "HTTP/1.0 503 Service Unavailable\r\nServer: Jabberwocky\r\nContent-Length: 15\r\n\r\nTry again later";
  62. static char NotFound[] = "HTTP/1.0 404 Not Found\r\nServer: Jabberwocky\r\nContent-Length: 14\r\n\r\nFile not found";
  63. static char Header[] = "HTTP/1.0 200 OK\r\nServer: Jabberwocky HTTPd 1.0\r\nContent-Type: application/octet-stream\r\n";
  64.  
  65. THREAD_DECL(http_agent);
  66. static int http_poll(struct connection *c);
  67. static void http_listen_msg(mthread *mt, int com, APTR data);
  68. static struct xferdata *http_match(char *name);
  69.  
  70.  
  71. int http_up(void)
  72. {
  73.     struct sockaddr_in sin;
  74.     long tmp = 1;
  75.  
  76.     if(sock != -1) return(0);
  77.  
  78.     if(!net_lala()) return(-1);
  79.  
  80.     sock = socket(AF_INET, SOCK_STREAM, 0);
  81.     if(sock < 0)
  82.     {
  83.         sock = -1;
  84.         net_bibi();
  85.         return -2;
  86.     }
  87.     setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &tmp, sizeof(tmp));
  88.  
  89.     memset(&sin, 0, sizeof(sin));
  90.     sin.sin_family = AF_INET;
  91.     sin.sin_port = prf.http_port;
  92.     if(bind(sock, (struct sockaddr *)&sin, sizeof(sin)) < 0)
  93.     {
  94.         CloseSocket(sock);
  95.         sock = -1;
  96.         net_bibi();
  97.         return -3;
  98.     }
  99.     listen(sock, 4);
  100.     IoctlSocket(sock, FIOASYNC, (char *)&tmp);
  101.  
  102. printf("listening...\n");
  103.     return 0;
  104. }
  105.  
  106.  
  107. void http_down(void)
  108. {
  109.     if(sock == -1) return;
  110.     CloseSocket(sock);
  111.     net_bibi();
  112.     sock = -1;
  113. }
  114.  
  115.  
  116. void http_listen(void)
  117. {
  118.     struct sockaddr_in sin;
  119.     long len, news;
  120.     struct connection *c;
  121.  
  122.     if(sock == -1) return;
  123.  
  124.     FD_ZERO(&fds);
  125.     FD_SET(sock, &fds);
  126.     if(WaitSelect(sock+1, &fds, NULL, NULL, &tv, 0) == 1)
  127.     {
  128.         len = sizeof(sin);
  129.         news = accept(sock, (struct sockaddr *)&sin, &len);
  130.         if(news >= 0)
  131.         {
  132. printf("new connection\n");
  133.             c = malloc(sizeof(struct connection));
  134.             if(c)
  135.             {
  136.                 c->key = 0;
  137.                 c->state = 0;
  138.                 c->sock = news;
  139.                 c->len = 0;
  140.                 c->buf[0] = '\0';
  141.                 list_append(&waitlist, c);
  142. printf("accepted\n");
  143.             }
  144.             else
  145.                 CloseSocket(news);
  146.         }
  147.     }
  148.  
  149.  
  150.     if(waitlist)
  151.     {
  152.         struct connection *c2;
  153.  
  154.         c = list_first(&waitlist);
  155.         while(c)
  156.         {
  157.             c2 = list_next(c);
  158.             if(c->state == 0 && http_poll(c) != 0)
  159.             {
  160.                 list_remove(&waitlist, c);
  161.                 free(c);
  162.             }
  163.             c = c2;
  164.         }
  165.     }
  166. }
  167.  
  168.  
  169. static int http_poll(struct connection *c)
  170. {
  171.     int tmp;
  172.  
  173.     FD_ZERO(&fds);
  174.     FD_SET(c->sock, &fds);
  175.     if(WaitSelect(c->sock+1, &fds, NULL, NULL, &tv, 0) == 1)
  176.     {
  177.         char *pika = c->buf;
  178.         pika += c->len;
  179.         tmp = recv(c->sock, pika, 1023 - c->len, 0);
  180.         if(tmp <= 0)
  181.         {
  182.             CloseSocket(c->sock);
  183.             return 1;
  184.         }
  185.         else
  186.         {
  187.             char *endval;
  188.             c->len += tmp;
  189.             c->buf[c->len] = '\0';
  190.             if(endval = strstr(c->buf, "\r\n\r\n"))
  191.             {
  192.                 char *buf = c->buf, *eol;
  193. printf("checking header\n");
  194.                 while(buf < endval)
  195.                 {
  196.                     if(isspace(buf[0]))
  197.                     {
  198.                         buf++;
  199.                         continue;
  200.                     }
  201.                     eol = strchr(buf, '\n');
  202.                     if(eol == NULL) eol = endval;
  203.  
  204.                     if(eol-buf > 4 && strncmp("GET ", buf, 4) == 0)
  205.                     {
  206.                         struct xferdata *xf;
  207.                         buf += 4;
  208.                         strtok(buf, " ");
  209.                         while(strchr(buf, '/')) buf = strchr(buf, '/') + 1;
  210.                         xf = http_match(buf);
  211.                         if(!xf)
  212.                         {
  213.                             send(c->sock, NotFound, sizeof(NotFound) - 1, 0);
  214.                             CloseSocket(c->sock);
  215.                             return 1;
  216.                         }
  217. printf("launching server thread\n");
  218.                         xf->state = XFER_CONNECTED;
  219.                         xfer_refresh(xf);
  220.                         c->xf = xf;
  221.                         c->key = ReleaseSocket(c->sock, GetUniqueID());
  222.                         c->state = 1;
  223.                         mt_run(&http_agent, c, http_listen_msg);
  224.                     }
  225.  
  226.                     buf = eol + 1;
  227.                 }
  228.             }
  229.         }
  230.     }
  231.     return 0;
  232. }
  233.  
  234.  
  235. static struct xferdata *http_match(char *name)
  236. {
  237.     struct xferdata *xf;
  238.  
  239.     if(!xfers || !name) return(NULL);
  240.     xf = list_first(&xfers);
  241.     while(xf)
  242.     {
  243.         if(strcmp(xf->name, name) == 0) return(xf);
  244.         xf = list_next(xf);
  245.     }
  246.     return NULL;
  247. }
  248.  
  249.  
  250. static void http_listen_msg(mthread *mt, int com, APTR data)
  251. {
  252.     struct connection *c = mt->data;
  253.  
  254.     switch(com)
  255.     {
  256.     case 42:
  257.         c->xf->state = XFER_ERROR;
  258. printf("error %d.\n", data);
  259.         xfer_refresh(c->xf);
  260.         break;
  261.  
  262.     case 41:
  263.         c->xf->state = XFER_SENDING;
  264.         xfer_refresh(c->xf);
  265.         break;
  266.  
  267.     case 40:
  268.         c->xf->state = XFER_FINISHED;
  269.         xfer_refresh(c->xf);
  270.         break;
  271.     }
  272. }
  273.  
  274.  
  275. #define ABORT(x) { mt_feedback(t, 42, (APTR) x ); goto out; }
  276.  
  277. THREAD(http_agent)
  278. {
  279.     mthread *t;
  280.     u_long sigs = 0, netsigmask = 0, msgsigmask;
  281.     struct Library *DOSBase = NULL;
  282.     struct Library *SocketBase = NULL;
  283.     struct sockaddr_in sin;
  284.     long sock = -1, len, tmp = 1, netsig = -1;
  285.     struct fd_set fds;
  286.     struct timeval tv = { 0, 0 };
  287.     struct connection *c;
  288.     BPTR f = NULL;
  289.     struct stat statusbuffer;
  290.  
  291.     t = mt_start();
  292.     if(!t) return;
  293.     c = t->data;
  294.  
  295.     netsig = AllocSignal(-1);
  296.     if(netsig == 1) ABORT(0);
  297.     netsigmask = 1L << netsig;
  298.     msgsigmask = 1L << (t->port->mp_SigBit);
  299.  
  300.     DOSBase = OpenLibrary("dos.library", 0);
  301.     if(!DOSBase) ABORT(1);
  302.  
  303.     SocketBase = OpenLibrary("bsdsocket.library", 0);
  304.     if(!SocketBase) ABORT(2);
  305.     SocketBaseTags(SBTM_SETVAL(SBTC_SIGIOMASK), (ULONG) netsigmask, TAG_DONE);
  306.  
  307.     sock = ObtainSocket((u_long)c->key, AF_INET, SOCK_STREAM, 0);
  308.     if(sock < 0) ABORT(3);
  309.  
  310.     f = Open(c->xf->file, MODE_OLDFILE);
  311.     if(!f) ABORT(4);
  312.  
  313.     send(sock, Header, sizeof(Header) - 1, 0);
  314.     
  315.     if (stat(c->xf->file, &statusbuffer) != -1) {
  316.         if (statusbuffer.st_size > 0) {
  317.             char length[36];
  318.             sprintf(length,"Content-Length: %u\r\n", statusbuffer.st_size);
  319.             send(sock, length, strlen(length), 0);
  320.         }
  321.     }
  322.  
  323.     send(sock, "\r\n", 2, 0);
  324.  
  325.     mt_feedback(t, 41, 0);
  326.     while(1)
  327.     {
  328.         sigs = SetSignal(0, netsigmask | msgsigmask | SIGBREAKF_CTRL_C);
  329.  
  330.         if(sigs & SIGBREAKF_CTRL_C)
  331.             break;
  332.  
  333.         len = Read(f, c->buf, 4095);
  334.         if(len == -1) ABORT(5);
  335.         if(len != 0)
  336.         {
  337.             send(sock, c->buf, len, 0);
  338.         }
  339.         if(len == 0) break;
  340.  
  341.         c->xf->cursize += len;
  342.     }
  343.     mt_feedback(t, 40, 0);
  344.  
  345. out:
  346.     if(f) Close(f);
  347.     if(sock != -1) CloseSocket(sock);
  348.     if(SocketBase) CloseLibrary(SocketBase);
  349.     if(DOSBase) CloseLibrary(DOSBase);
  350.     if(netsig != -1) FreeSignal(netsig);
  351.     mt_end(t);
  352. }
  353.